import torch
# pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

print("torch version:",torch.version.__version__)
print("cuda version:",torch.version.cuda)
print("is_available:",torch.cuda.is_available())
print("device_count:",torch.cuda.device_count())
print("cudnn version:",torch.backends.cudnn.version())
try:
    # 按照流程安装 \TensorRT.md
    import tensorrt
    print('tensorrt:'+tensorrt.__version__)
except ImportError:
    print("TensorRT未安装,请安装TensorRT")